#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from generate.sub.itjuzi import ItjuziProjectGenerator
from generate.sub.qhee import QheeProjectListGenerator, QheeProjectGenerator
from generate.sub.smt import SmtProjectGenerator, SmtInfoBaseProjectGenerator, SmtInfoListProjectGenerator
from generate.sub.zhiXiaoBin import zhiXiaoBinProjectGenerator
from handle.sub.itjuzi import ItJuZiProjectHandler
from handle.sub.smt import SmtProjectHandler, SmtInfoBaseProjectHandler, SmtInfoListProjectHandler
from handle.sub.cyzone import CyzoneProjectListHandler, CyzoneProjectHandler
from generate.sub.cyzone import CyzoneProjectListGenerator, CyzoneProjectGenerator


# # 生成任务
# python3 main.py download_cyzone_project_list generator
# # 下载任务
# python3 main.py download_cyzone_project_list handler
# {task_type: {action_type: deal_class}
from handle.sub.qhee import QheeProjectListHandler, QheeProjectHandler
from handle.sub.zhiXiaoBin import zhiXiaoBinProjectHandler


def get_task_dic():
    return {
        'download_cyzone_project': {
            'handler': CyzoneProjectHandler,  # 下载任务
            'generator': CyzoneProjectGenerator  # 生成任务
        },
        'download_cyzone_project_list': {
            'handler': CyzoneProjectListHandler,
            'generator': CyzoneProjectListGenerator
        },
        'download_qhee_project_list': {
            'handler': QheeProjectListHandler,
            'generator': QheeProjectListGenerator
        },
        'download_qhee_project': {
            'handler': QheeProjectHandler,
            'generator': QheeProjectGenerator
        },
        'download_smt_project': {
            'handler': SmtProjectHandler,
            'generator': SmtProjectGenerator
        },
        'download_smt_info_list_project': {
            'handler': SmtInfoListProjectHandler,
            'generator': SmtInfoListProjectGenerator
        },
        'download_smt_info_base_project': {
            'handler': SmtInfoBaseProjectHandler,
            'generator': SmtInfoBaseProjectGenerator
        },
        'download_zhixiaobing_company': {
            'handler': zhiXiaoBinProjectHandler,
            'generator': zhiXiaoBinProjectGenerator
        },
        'download_itjuzi_project': {
            'handler': ItJuZiProjectHandler,
            'generator': ItjuziProjectGenerator
        }
    }
